package com.atwy.iterator;

import java.util.Arrays;

public class TestNestedIterator {
    public static void main(String[] args) {
        //输入：nestedList = [[1,1],2,[1,1]] 输出：[1,1,2,1,1]
        // 测试扁平化嵌套列表迭代器
        NestedInteger n1 = new NestedIntegerImpl(1);
        NestedInteger n2 = new NestedIntegerImpl(1);
        NestedInteger n3 = new NestedIntegerImpl(2);
        NestedInteger n4 = new NestedIntegerImpl(1);
        NestedInteger n5 = new NestedIntegerImpl(1);

        NestedInteger n11 = new NestedIntegerImpl(Arrays.asList(n1,n2));
        NestedInteger n22 = new NestedIntegerImpl(Arrays.asList(n4,n5));


        // 测试脚本
        NestedIterator iterator = new NestedIterator(Arrays.asList(n11,n3,n22));
        while (iterator.hasNext()){
            System.out.println(iterator.next());
        }
    }
}
